package socket;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.nio.charset.StandardCharsets;
import java.util.ArrayList;
import java.util.List;

/**
 * 聊天室的服务端 版本5
 *
 * 完善服务端的功能
 * 1:添加广播某个客户端上下线的情况(并且广播当前在线人数)
 *
 *
 *
 * 实现:
 * 功能1:广播消息
 *  1:将广播消息的操作提取成一个方法,并取名为sendMessage
 *  2:在线程启动后(ClientHandler的run方法中)当我们将该客户端输出流存入共享集合allOut后
 *    就广播给所有客户端该用户上线了,并且利用allOut集合的size作为当前在线人数.
 *  3:在run方法的异常处理机制最后添加finally,用于处理客户端断开后的操作
 *    3.1:将PrintWriter从try中定义改为在try之前定义,提高它的作用域,以便在finally中将其从
 *        allOut集合中移除
 *    3.2:广播消息给所有客户端告知该客户端下线了.
 *
 */
public class Server5 {
    /*
      java.net.ServerSocket
      运行在服务端的ServerSocket的主要任务:
      1.向服务端申请服务端口，客户端就是通过该端口与服务器建立连接的
      2.监听服务端口，一旦一个客户端连接了就会立即创建一个Socket实例与之交互
        我们将ServerSocket比喻为某客服中心的"总机"
     */
    private ServerSocket serverSocket;

    //V4:该集合用于保存所有客户端的输出流,用于广播消息
    private List<PrintWriter> allOut = new ArrayList<>();


    public Server5(){
        try {
            /*
              如果端口被其他程序占用了，则会发生java.net.BindException异常: Address already in use
              解决办法:
              1.更换端口号
              2.杀死占用该端口的进程(通常发生在服务端启动两次导致的情况下)
             */
            System.out.println("正在启动服务端");
            serverSocket = new ServerSocket(8088);
            System.out.println("服务端启动完毕!");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public void start(){
        /*
           ServerSocket提供的重要方法:
           Socket accept()
           用于接收客户端的连接，并返回一个Socket实例与连接的客户端交互
           该方法是一个阻塞方法，调用后程序会"卡住"，直到一个客户端连接为止
         */
        try {
            while(true){
                System.out.println("等待客户端连接");
                Socket socket = serverSocket.accept(); //接电话
                System.out.println("一个客户端连接上了");
                //启动一个线程负责与该客户端交互
                ClientHandler handler = new ClientHandler(socket);
                Thread t = new Thread(handler);
                t.start();
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        Server5 server = new Server5();
        server.start();
    }

    /**
     * 该线程任务负责与指定的客户端交互
     */
    private class ClientHandler implements Runnable{

        private Socket socket;
        private String host;//记录该客户端的IP地址

        public ClientHandler(Socket socket){
            this.socket = socket;
            //通过socket获取远端计算机的IP地址的字符串格式
            host = socket.getInetAddress().getHostAddress();
        }

        public void run() {
            //V5:提高pw的作用域,以便在finally中可以将其从allOut集合中删除
            PrintWriter pw = null;
            try {
                InputStream in = socket.getInputStream();
                InputStreamReader isr = new InputStreamReader(in, StandardCharsets.UTF_8);
                BufferedReader br = new BufferedReader(isr);

                OutputStream out = socket.getOutputStream();
                OutputStreamWriter osw = new OutputStreamWriter(out, StandardCharsets.UTF_8);
                BufferedWriter bw = new BufferedWriter(osw);
                pw = new PrintWriter(bw,true);

                //V5:多线程向集合中添加元素可能产生并发安全问题,因此必须同步执行
                synchronized (allOut) {//通常同步监视器对象就选取临界资源
                    allOut.add(pw);
                }
                //v5:广播消息,该客户端上线了
                sendMessage(host+"上线了,当前在线人数:"+allOut.size());


                String message;
                while((message=br.readLine())!=null){
                    //V5:读取到客户端消息广播时改为调用sendMessage方法
                    sendMessage(host + "说:" + message);
                }
            } catch (IOException e) {
            //V5:在finally中处理客户端断开后的操作
            } finally {
                //将该客户端的输出流从allOut集合中删除
                synchronized (allOut) {
                    allOut.remove(pw);
                }
                //广播该客户端下线了
                sendMessage(host+"下线了,当前在线人数:"+allOut.size());
                try {
                    //关闭socket,释放资源
                    socket.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }

        /**
         * V5的功能一,实现广播消息的方法:
         * 将消息转发给所有客户端
         * @param message
         */
        private void sendMessage(String message){
            System.out.println(message);
            //V5:新循环就是迭代器,迭代器要求遍历过程不能用集合方法增删元素,所以需要和上面的增删互斥
            synchronized (allOut) {
                for (PrintWriter o : allOut) {
                    o.println(message);
                }
            }
        }


    }

}

















